import matplotlib.pyplot as plt
from matplotlib.pylab import mpl
import numpy as np
from numpy import pi, cos, sin

# 中文显示问题
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False


'''
一维时序信号y，它由5V的直流分量(0Hz)，和振幅为2V，频率为20Hz的交流信号，以及振幅为1V，频率为160Hz的交流信号组成。
'''

fs = 800  # 采样频率， 要大于信号频率的两倍
t = np.arange(0, 1, 1.0 / fs)  # 1秒采样fs个点
N = len(t)
freq = np.arange(N)  # 频率counter

x = 5 + 2 * cos(2 * pi * 20 * t) + 1 * cos(2 * pi * 160 * t)  # 离散化后的x[n]

print(x)


# 曲线图
plt.subplot(2, 1, 1)
plt.plot(t, x)
plt.title('曲线图')
# plt.xlabel('Time(s)')
plt.ylabel('Amplitude')

# 散点图
plt.subplot(2, 1, 2)
plt.plot(t, x, 'go')
plt.title('散点图')
plt.xlabel('Time(s)')
plt.ylabel('Amplitude')

plt.show()
